#ifndef GPU_H
#define GPU_H

#include <iostream>
#include <vector>
#include <string>
#include <cuda_runtime.h>

// 前向声明，而不是包含整个头文件
class PT;
class model;

// GPU函数声明：处理单段PT的猜测生成

//我们要计算所有pt的概率值
void GPU_CalProb(std::vector<PT> &new_pts, model &m);

#endif // GPU_H